# Databricks notebook source
# MAGIC %md
# MAGIC # Installations

# COMMAND ----------

# MAGIC %pip install -U mostlyai[local] 
# MAGIC %pip install ipywidgets
# MAGIC %restart_python

# COMMAND ----------

import time
time.sleep(600)

# COMMAND ----------

import pandas as pd
from mostlyai.sdk import MostlyAI

mostly = MostlyAI(local=True)

# COMMAND ----------

# MAGIC %md
# MAGIC # Get generatior

# COMMAND ----------

[print(i) for i in mostly.generators.list()]

# COMMAND ----------

generator = mostly.generators.get("c5528a9d-7991-47e0-aa93-935dabc71484")

# COMMAND ----------

# MAGIC %md
# MAGIC ## Save generator

# COMMAND ----------

generator.export_to_file('/Volumes/admin_govern_sta_des/tmp_mostlyai/generator/tSocial_31_03.zip')

# COMMAND ----------

# MAGIC %md
# MAGIC # Generation

# COMMAND ----------

df_samples = mostly.generate(generator, size=87_000)